Node / meshtastic / Meshtastic-Android / files / androidApp / src / google / kotlin / org / meshtastic / app / translation / MlKitDocTranslator.kt

Displaying Raw • Download

T8b949e/*

* Copyright (c) 2026 Meshtastic LLC

*

* This program is free software: you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation, either version 3 of the License, or

* (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program. If not, see <https://www.gnu.org/licenses/>.

*/

Tff7b72package T7ee787org.meshtastic.app.translation

Tff7b72import T7ee787co.touchlab.kermit.Logger

Tff7b72import T7ee787com.google.mlkit.common.model.DownloadConditions

Tff7b72import T7ee787com.google.mlkit.common.model.RemoteModelManager

Tff7b72import T7ee787com.google.mlkit.nl.translate.TranslateLanguage

Tff7b72import T7ee787com.google.mlkit.nl.translate.TranslateRemoteModel

Tff7b72import T7ee787com.google.mlkit.nl.translate.Translation

Tff7b72import T7ee787com.google.mlkit.nl.translate.TranslatorOptions

Tff7b72import T7ee787kotlinx.coroutines.suspendCancellableCoroutine

Tff7b72import T7ee787org.meshtastic.feature.docs.translation.DocTranslationCache

Tff7b72import T7ee787org.meshtastic.feature.docs.translation.DocTranslationService

Tff7b72import T7ee787org.meshtastic.feature.docs.translation.DownloadResult

Tff7b72import T7ee787org.meshtastic.feature.docs.translation.MarkdownTranslationSegmenter

Tff7b72import T7ee787org.meshtastic.feature.docs.translation.TranslationResult

Tff7b72import T7ee787org.meshtastic.feature.docs.translation.md5Hash

Tff7b72import T7ee787kotlin.coroutines.resume

T8b949e/**

* ML Kit-powered document translation service for the Google flavor.

*

* Downloads language models on-demand (~30MB each) and translates markdown content while preserving structure via

* [MarkdownTranslationSegmenter].

*/

Tff7b72class T56d364MlKitDocTranslatorTb4b4b4(Tff7b72private Tff7b72val Te6edf3cacheTb4b4b4: Te6edf3DocTranslationCacheTb4b4b4) Tb4b4b4: Te6edf3DocTranslationService Tb4b4b4{

Tff7b72private Tff7b72val Te6edf3modelManager Tff7b72= Te6edf3RemoteModelManagerTb4b4b4.Te6edf3getInstanceTb4b4b4(Tb4b4b4)

Tff7b72override Tff7b72suspend Tff7b72fun Td2a8fftranslatePageTb4b4b4(Te6edf3pageIdTb4b4b4: Tffa657StringTb4b4b4, Te6edf3markdownTb4b4b4: Tffa657StringTb4b4b4, Te6edf3targetLocaleTb4b4b4: Tffa657StringTb4b4b4)Tb4b4b4: Te6edf3TranslationResult Tb4b4b4{

Tff7b72val Te6edf3sourceHash Tff7b72= Te6edf3md5HashTb4b4b4(Te6edf3markdownTb4b4b4)

T8b949e// Check cache first

Te6edf3cacheTb4b4b4.Te6edf3getTb4b4b4(Te6edf3pageIdTb4b4b4, Te6edf3targetLocaleTb4b4b4, Te6edf3sourceHashTb4b4b4)Tff7b72?.Te6edf3let Tb4b4b4{ Te6edf3cached Tff7b72-Tff7b72>

Tff7b72return Te6edf3TranslationResultTb4b4b4.Te6edf3SuccessTb4b4b4(Te6edf3cachedTb4b4b4)

Tb4b4b4}

T8b949e// Check if language is supported by ML Kit

Tff7b72val Te6edf3targetLang Tff7b72= Te6edf3TranslateLanguageTb4b4b4.Te6edf3fromLanguageTagTb4b4b4(Te6edf3targetLocaleTb4b4b4) Tff7b72?: Tff7b72return Te6edf3TranslationResultTb4b4b4.Te6edf3Unavailable

T8b949e// Auto-download model if not present

Tff7b72if Tb4b4b4(Tff7b72!Te6edf3isModelDownloadedTb4b4b4(Te6edf3targetLangTb4b4b4)Tb4b4b4) Tb4b4b4{

Te6edf3LoggerTb4b4b4.Te6edf3iTb4b4b4(Te6edf3tag Tff7b72= Ta5d6ff"Ta5d6ffMlKitDocTranslatorTa5d6ff"Tb4b4b4) Tb4b4b4{

Ta5d6ff"Ta5d6ffDownloading model for Tffd700$Te6edf3targetLocaleTa5d6ff (~Tffd700${Te6edf3ESTIMATED_MODEL_SIZE_MBTffd700}Ta5d6ffMB)Ta5d6ff"

Tb4b4b4}

Tff7b72val Te6edf3downloadResult Tff7b72= Te6edf3downloadLanguageModelTb4b4b4(Te6edf3targetLocaleTb4b4b4)

Tff7b72if Tb4b4b4(Te6edf3downloadResult Tff7b72is Te6edf3DownloadResultTb4b4b4.Te6edf3FailedTb4b4b4) Tb4b4b4{

Te6edf3LoggerTb4b4b4.Te6edf3wTb4b4b4(Te6edf3tag Tff7b72= Ta5d6ff"Ta5d6ffMlKitDocTranslatorTa5d6ff"Tb4b4b4) Tb4b4b4{ Ta5d6ff"Ta5d6ffModel download failed: Tffd700${Te6edf3downloadResultTb4b4b4.Te6edf3reasonTffd700}Ta5d6ff" Tb4b4b4}

Tff7b72return Te6edf3TranslationResultTb4b4b4.Te6edf3ModelDownloadRequiredTb4b4b4(Te6edf3targetLocaleTb4b4b4, Te6edf3ESTIMATED_MODEL_SIZE_MBTb4b4b4)

Tb4b4b4}

Tb4b4b4}

T8b949e// Perform translation

Tff7b72return Tff7b72try Tb4b4b4{

Tff7b72val Te6edf3options Tff7b72=

Te6edf3TranslatorOptionsTb4b4b4.Te6edf3BuilderTb4b4b4(Tb4b4b4)

Tb4b4b4.Te6edf3setSourceLanguageTb4b4b4(Te6edf3TranslateLanguageTb4b4b4.Te6edf3ENGLISHTb4b4b4)

Tb4b4b4.Te6edf3setTargetLanguageTb4b4b4(Te6edf3targetLangTb4b4b4)

Tb4b4b4.Te6edf3buildTb4b4b4(Tb4b4b4)

Tff7b72val Te6edf3translator Tff7b72= Te6edf3TranslationTb4b4b4.Te6edf3getClientTb4b4b4(Te6edf3optionsTb4b4b4)

Tff7b72try Tb4b4b4{

Tff7b72val Te6edf3translated Tff7b72=

Te6edf3MarkdownTranslationSegmenterTb4b4b4.Te6edf3translateMarkdownTb4b4b4(Te6edf3markdownTb4b4b4) Tb4b4b4{ Te6edf3text Tff7b72-Tff7b72>

Te6edf3suspendCancellableCoroutine Tb4b4b4{ Te6edf3cont Tff7b72-Tff7b72>

Te6edf3translator

Tb4b4b4.Te6edf3translateTb4b4b4(Te6edf3textTb4b4b4)

Tb4b4b4.Te6edf3addOnSuccessListener Tb4b4b4{ Te6edf3contTb4b4b4.Te6edf3resumeTb4b4b4(Tffa657itTb4b4b4) Tb4b4b4}

Tb4b4b4.Te6edf3addOnFailureListener Tb4b4b4{ Te6edf3e Tff7b72-Tff7b72>

Te6edf3LoggerTb4b4b4.Te6edf3wTb4b4b4(Te6edf3tag Tff7b72= Ta5d6ff"Ta5d6ffMlKitDocTranslatorTa5d6ff"Tb4b4b4) Tb4b4b4{

Ta5d6ff"Ta5d6ffSegment translation failed, using source: Tffd700${Te6edf3eTb4b4b4.Te6edf3messageTffd700}Ta5d6ff"

Tb4b4b4}

Te6edf3contTb4b4b4.Te6edf3resumeTb4b4b4(Te6edf3textTb4b4b4)

Tb4b4b4}

Tb4b4b4}

Tb4b4b4}

T8b949e// Cache the result

Te6edf3cacheTb4b4b4.Te6edf3putTb4b4b4(Te6edf3pageIdTb4b4b4, Te6edf3targetLocaleTb4b4b4, Te6edf3sourceHashTb4b4b4, Te6edf3translatedTb4b4b4)

Te6edf3TranslationResultTb4b4b4.Te6edf3SuccessTb4b4b4(Te6edf3translatedTb4b4b4)

Tb4b4b4} Tff7b72finally Tb4b4b4{

Te6edf3translatorTb4b4b4.Te6edf3closeTb4b4b4(Tb4b4b4)

Tb4b4b4}

Tb4b4b4} Tff7b72catch Tb4b4b4(Te6edf3eTb4b4b4: Te6edf3ExceptionTb4b4b4) Tb4b4b4{

Te6edf3LoggerTb4b4b4.Te6edf3wTb4b4b4(Te6edf3tag Tff7b72= Ta5d6ff"Ta5d6ffMlKitDocTranslatorTa5d6ff"Tb4b4b4) Tb4b4b4{ Ta5d6ff"Ta5d6ffTranslation failed for Tffd700$Te6edf3pageIdTa5d6ff to Tffd700$Te6edf3targetLocaleTa5d6ff: Tffd700${Te6edf3eTb4b4b4.Te6edf3messageTffd700}Ta5d6ff" Tb4b4b4}

Te6edf3TranslationResultTb4b4b4.Te6edf3Unavailable

Tb4b4b4}

Tb4b4b4}

Tff7b72override Tff7b72suspend Tff7b72fun Td2a8ffisLanguageAvailableTb4b4b4(Te6edf3localeTb4b4b4: Tffa657StringTb4b4b4)Tb4b4b4: Tffa657Boolean Tb4b4b4{

Tff7b72val Te6edf3lang Tff7b72= Te6edf3TranslateLanguageTb4b4b4.Te6edf3fromLanguageTagTb4b4b4(Te6edf3localeTb4b4b4) Tff7b72?: Tff7b72return Tff7b72false

Tff7b72return Te6edf3isModelDownloadedTb4b4b4(Te6edf3langTb4b4b4)

Tb4b4b4}

Tff7b72override Tff7b72suspend Tff7b72fun Td2a8ffdownloadLanguageModelTb4b4b4(Te6edf3localeTb4b4b4: Tffa657StringTb4b4b4)Tb4b4b4: Te6edf3DownloadResult Tb4b4b4{

Tff7b72val Te6edf3lang Tff7b72=

Te6edf3TranslateLanguageTb4b4b4.Te6edf3fromLanguageTagTb4b4b4(Te6edf3localeTb4b4b4) Tff7b72?: Tff7b72return Te6edf3DownloadResultTb4b4b4.Te6edf3FailedTb4b4b4(Ta5d6ff"Ta5d6ffUnsupported language: Tffd700$Te6edf3localeTa5d6ff"Tb4b4b4)

Tff7b72val Te6edf3model Tff7b72= Te6edf3TranslateRemoteModelTb4b4b4.Te6edf3BuilderTb4b4b4(Te6edf3langTb4b4b4)Tb4b4b4.Te6edf3buildTb4b4b4(Tb4b4b4)

Tff7b72val Te6edf3conditions Tff7b72= Te6edf3DownloadConditionsTb4b4b4.Te6edf3BuilderTb4b4b4(Tb4b4b4)Tb4b4b4.Te6edf3buildTb4b4b4(Tb4b4b4)

Tff7b72return Te6edf3suspendCancellableCoroutine Tb4b4b4{ Te6edf3cont Tff7b72-Tff7b72>

Te6edf3modelManager

Tb4b4b4.Te6edf3downloadTb4b4b4(Te6edf3modelTb4b4b4, Te6edf3conditionsTb4b4b4)

Tb4b4b4.Te6edf3addOnSuccessListener Tb4b4b4{ Te6edf3contTb4b4b4.Te6edf3resumeTb4b4b4(Te6edf3DownloadResultTb4b4b4.Te6edf3SuccessTb4b4b4) Tb4b4b4}

Tb4b4b4.Te6edf3addOnFailureListener Tb4b4b4{ Te6edf3e Tff7b72-Tff7b72> Te6edf3contTb4b4b4.Te6edf3resumeTb4b4b4(Te6edf3DownloadResultTb4b4b4.Te6edf3FailedTb4b4b4(Te6edf3eTb4b4b4.Te6edf3message Tff7b72?: Ta5d6ff"Ta5d6ffDownload failedTa5d6ff"Tb4b4b4)Tb4b4b4) Tb4b4b4}

Tb4b4b4}

Tb4b4b4}

Tff7b72private Tff7b72suspend Tff7b72fun Td2a8ffisModelDownloadedTb4b4b4(Te6edf3langTb4b4b4: Tffa657StringTb4b4b4)Tb4b4b4: Tffa657Boolean Tff7b72= Te6edf3suspendCancellableCoroutine Tb4b4b4{ Te6edf3cont Tff7b72-Tff7b72>

Tff7b72val Te6edf3model Tff7b72= Te6edf3TranslateRemoteModelTb4b4b4.Te6edf3BuilderTb4b4b4(Te6edf3langTb4b4b4)Tb4b4b4.Te6edf3buildTb4b4b4(Tb4b4b4)

Te6edf3modelManager

Tb4b4b4.Te6edf3isModelDownloadedTb4b4b4(Te6edf3modelTb4b4b4)

Tb4b4b4.Te6edf3addOnSuccessListener Tb4b4b4{ Te6edf3contTb4b4b4.Te6edf3resumeTb4b4b4(Tffa657itTb4b4b4) Tb4b4b4}

Tb4b4b4.Te6edf3addOnFailureListener Tb4b4b4{ Te6edf3contTb4b4b4.Te6edf3resumeTb4b4b4(Tff7b72falseTb4b4b4) Tb4b4b4}

Tb4b4b4}

Tff7b72companion Tff7b72object Tb4b4b4{

Tff7b72private Tff7b72const Tff7b72val Te6edf3ESTIMATED_MODEL_SIZE_MB Tff7b72= T79c0ff3T79c0ff0

Tb4b4b4}

Tb4b4b4}